package com.ruoyi.constant;

import com.volcengine.ark.runtime.model.completion.chat.ChatMessageRole;

import java.util.HashMap;
import java.util.Map;

public class AiFinalConfig {

    /**
     * ai角色字典标识
     */
    public final static Map<String,ChatMessageRole> aiRole= new HashMap<String,ChatMessageRole>();
    static{
        aiRole.put("user", ChatMessageRole.USER);//用户消息
        aiRole.put("system",ChatMessageRole.SYSTEM);//系统消息
        aiRole.put("assistant",ChatMessageRole.ASSISTANT);//对话助手消息
    }
}
